%run train-cGAN # %run unit_test
self.device is cuda:0 64 Epoch: 0, Itration: 0 loss_G_GAN = 4.886846542358398, loss_G_L1 = 0.39250922203063965 * 5 loss_D = 0.7143234014511108
Epoch: 76, Itration: 1000 loss_G_GAN = 9.659367561340332, loss_G_L1 = 0.15902411937713623 * 5 loss_D = 0.006910865660756826
Epoch: 153, Itration: 2000 loss_G_GAN = 8.78011417388916, loss_G_L1 = 0.17440780997276306 * 5 loss_D = 0.0005042690900154412
save model to ./checkpoints/pix2food.pkl at iteration = 3000 Epoch: 230, Itration: 3000 loss_G_GAN = 4.836752891540527, loss_G_L1 = 0.17835037410259247 * 5 loss_D = 0.18689367175102234
Finally save model to ./checkpoints/pix2food.pkl # --- Validation Set --- #
# %run train-cGAN
# %run train